import mnist
import viz